from django.shortcuts import render


import logging
from django.views import View
from django import http
from django.core.cache import cache

from areas.models import Area
from meiduo_mall.utils.response_code import RETCODE
# Create your views here.
logger = logging.getLogger('django')


class AreasView(View):
    """省市区数据"""

    def get(self, request):
        # 判断当前是要查询省份数据还是市区数据
        # 如果前端没有传入 area_id，表示用户需要省份数据
        # 如果前端传入了 area_id，表示用户需要市或区数据
        area_id = request.GET.get('area_id')
        if not area_id:
            province_list = cache.get('province_list')
            # 查询是否有缓存
            if not province_list:
                # 查询省份数据
                try:
                    province_list = list(Area.objects.filter(parent__isnull=True).values('id', 'name'))
                except Exception as e:
                    logger.error(e)
                    return http.JsonResponse({'code': RETCODE.DBERR, 'errmsg': '查询省份数据错误'})
                cache.set('province_list', province_list, 3600)
            return http.JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'province_list': province_list})

        else:
            sub_data = cache.get('sub_area_' + area_id)
            if not sub_data:
                try:
                    # 查询市或区的父级
                    parent_model = Area.objects.get(id=area_id)
                    # sub_model_list = parent_model.<关联字段>_set.all()
                    # 由于模型类定义的 related_name='subs', 故不能这么写
                    subs = list(parent_model.subs.all().values('id', 'name'))

                    sub_data = {
                        'id': parent_model.id,  # 父级id
                        'name': parent_model.name,  # 父级name
                        'subs': subs # 父级的子集
                    }
                except Exception as e:
                    logger.error(e)
                    return http.JsonResponse({'code': RETCODE.DBERR, 'errmsg': '获取城市或区县数据错误'})
                cache.set('sub_area_' + area_id, sub_data, 3600)
            # 响应城市或区县JSON数据
            return http.JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'sub_data': sub_data})
